export MODEL_NAME='/mnt/afs2d/luotianhang/cache/PretrainedModels/stable-diffusion-v1-5/stable-diffusion-v1-5'
export OUTPUT_DIR="savemodel"

accelerate launch train_controlnet.py \
 --pretrained_model_name_or_path=$MODEL_NAME \
 --output_dir=/mnt/afs2d/luotianhang/smartvehicle_diffusion/diffusers/examples/controlnet/controlnet_experiment_fill50k_4 \
 --dataset_name=fusing/fill50k \
 --resolution=512 \
 --learning_rate=1e-5 \
 --validation_image "./conditioning_image_1.png" "./conditioning_image_2.png" \
 --validation_prompt "red circle with blue background" "cyan circle with brown floral background" \
 --train_batch_size=4